square a number in python 3